

def save(graph,path):
    graph_image = graph.get_graph().draw_mermaid_png()
    image_path = path
    with open(image_path, "wb+") as f:
        f.write(graph_image)